Added localStorage for persistence