TODO: add UI logic for UIButton