Save every step, show models in display.py