incorporate these changes to complete assignment

incorporate these changes to complete assignment.

thx