Add persistence via localStorage