Added python code for the project