Working on the simple problem from part 1