changed state_machine.py to save the current output and not write it to None every step