Refactored UI for docs