Got part 1 working