Bug/junit imports

hi! just one small change. After merging pull request #75 the project wasn't compiling because of some missing junit imports.