working on 1-2
problem 1-2 working