Add saving if save_time_step is used. Close #19
Add sensitivity saving on last time step. Fix #18