Adding loading to buttons