refactored feature test followed the solution video