added reinforce logic for ui