updates to add save method