Completed the Unit Tests assignment