Adding assignment 6 solution