Made the state section more clear