Cleaning up code for task 2 with updated Readme.