use org.junit package
get rid of duplicated code