Made save and load work