Improve documentation for UI classes