Added py file for Sheet 4 Exercise 2