Root Inc. interview question

Add a simple feature to the coding challenge that you completed.