update notebook interface based on problems at training