Improve docs UI