adding changes from work with assertion graph widget