Create a test class using Junit