Got problem 3 done