Refactored python code.
Removed outdated class files
Refactored python code