Code now formatted properly to PEP 8 standards