did some minor changes, can't remember what exactly, something to do with local storage