Cancel button now works