Completed code for the first feature test