Added code for a fun little problem.