adding a python file that does the same thing as the .cpp file