updated assignment 1