Added a python file for assignment 2.