Updated after running JUnit