* fix(ui): some ui reviews and fixes for feature flags * feat(ui): added some tests * feat(ui): added some tests