Add solution to #3