My solution of assignment2

Modified the original file and completed this assignment