Toggle button now works