starting to implement the model with three states