Using now junit 4 for test class