Add solution to problem 001