Made changes to complete assignment 2.