Working on comparing states correctly.