Use list comprehensions instead of separate vars

This saves space by using Python's super-awesome [List Comprehensions](https://docs.python.org/2/tutorial/datastructures.html#list-comprehensions)
Refactor return of booleans from if block

Here we just return the result of the test itself.
remove indentation from otherwise-blank lines
Merge pull request #2 from NotBobTheBuilder/minor-refactoring

Minor refactoring