Expose buttons and switch to UIButton class