Improve Problem 3 solution
Work in progress on Problem 33