added unit testing homework