Ide improvements
PEP8 and formatting fixes