Add option to enable local storage