work done

Edited the file which does the job that the assignment needs.