Got problem 1 working