Add org.junit for unit tests