To be consistent, put this into the setState function (don't forget about the lines below this as well, making changes as necessary)