Writing application code to fix the test – GREEN