Modify import to use org.junit.Assert rather than deprecated junit.framework.Assert