Changed SetupState to look prettier.