Updated button ref