updated assignment1 menu