Fix formatting bug, PEP8 formatting