Added a python script for worksheet 3