Adding the Junit Tests