First test passing, but need more to it.