finished problem 1 working on 2