Added simple solution - TDD Style