Complete save and load functionality