add unit testing to assignment 5 to repo