Implemented Save and load function, along with some minor refractoring