Fixed all PEP warnings (thanks to IntelliJ)