Challenge #2 - added unit testing to app