WIP: Problem 3 solution