Code still doesnt work, but at least it should now be PEP-8 compatible