Adding save to localStorage feature