PEP8 + Bug fixes