Make save method more versatile

Make it work with updates too, not only creations.