Add shouldComponentUpdate method