Add another test, make it pass