Add support for local storage