CUI updates

Some tidying up of CUI