Improve AI appearance, clear button