Updated assignment 2 with comments