Completing code for a unit tested case