(#7) - still trying to get tests to pass