Fixed initial tests.
The one function is now passing.