Added function to calculate game state (index in Q-table)
Improved debugging and diagrams

- If nr_trials is 1, then prints out all intermediate actions and
matrices.
- Added errorbar plot.