Working on shouldComponentUpdate