Added code for assignment.