Add the homework 5 solution