Add Storage class for persistence