Working on assignment 6
Finishing assignment 6