now with dropout and saving