Modify imports to use non-deprecated org.junit rather than deprecated junit.framework