Button now works properly