Added rest of project. Seperated Python functionality to a class