#4 Implement solution to exercise