Still trying to get the toggle to work