cleaned import, added  junit dependency explicitely