Fixing the models and adding functionalityt o create new "states" (ie actions)