Got the test to run and pass