Better comment
Some space changes to keep up with PEP8