Add solution for homework 3&4