Code modified during interview

The code modified during the follow-up interview after original solution was submitted.

Original submission

Original code submitted as a part the of interview process. It was later reviewed and modified during the interview that followed the submission.