Save button now works