code clean up for assignment 2 and updated readme