updated assignment with corrections