Add more tests, and pass it.