add update logic for training
update GUI for cleanning module