Made it so you can save and continue editing