Adding class for States