Added another function and two more tests that need to pass.
Tweaking test