Now I am PEP8 complient (I'll squash later)