Improved the project by adding a python file