save button now works fine