Add dependency to junit
Add assert import