Update success function within controller