Adding UI modifications