Added 4th test, changed output