Add support for localStorage