reworked ui module extension