PEP8 + removed unused imports + moved to a proper UnitTest library