Submitting correct version

@JamieMihoko as you guessed, there is a simpler way to do this, and it's with list comprehensions... you can use code very similar to the code from my last comment.