Add localStorage support