add assignment 5 solution