working on UI button states