edited next_step method in story tree