Button is now working