Shouldn't we manage this states with class attributhes rather than strings?