done the jUnit code