Enhancements to AI Studio