updated assignment 5