Added functionalities : save and load