Add assist functionality