Updated save function!
Updated save and load function to be functional.