Minor change in Save method