Adding the python script (not completed)