JUnit Exercise for CS-401