Completed assignment and added unit tests