Add fn comments to soln for "Falling Diamonds"