refactor python classes into seperate directory