Junit task

Please review