Improve stats collection

* Find best score for epoch
* Count number of simulations run for better accuracy