Updated UI

Added button focus for when button action is selected.