Added train run editing