Add state class