improving ui functionality