Working on getting the function test to pass