Add solution to problem #3
Add solution to problem #4