Added comments, conformed to PEP

Just a bit of cleaning up the code.