assignment 2 updated