[#44] Added context_ui into the mix