Commenting and making things nicer (PEP8)