Endava interview question

Add functionality to existing codebase