TODO: Fix local data!!
Fixed local storage bug