Improvements for assistant workflow