text: **Figure 5. A**learning for the actual reward condition (reward condition left). The agent correctly 
  learned the probability of receiving rewards in the rewarding arm. It did not learn the probabilities of 
  the non-rewarding arm since it did not explore that option. The color grading signifies the likelihood 
  of an observation being generated by a specific state. The more saturated the color, the higher 
  the likelihood. 
  
  4.3. Fitting the Model to the Data 