Add code for passing tests in score method