Added controls for save and load