I just moved the save from the above methods. See below where it's removed.