Added UIButton into the mix as well.