Begin clean and the code (PEP8, remove import unused, refactoring in code)