update ui module and new extension classes