* add earlystopping callback * addressing comments * address comments * addressing comments * remove unused imports