Add state generator