Making button states save to localStorage