Use state and props correctly. I think