fixed imports. Proper jUnit using now