udpated the assignment