A little list comprehension should clear things up I think. //