Add solution to problem #1