Updated docs and UI