Still trying to fix this damn button