Code styling

PEP8 and stuff