Tests updated for junit4