Add save method