Add solution for homework