Worked on the Junit Tests