Yes, it's a good idea and is already on my list for further improvements (on_loadstep function).