Couple of changes within the button