Updated code to work with built in nengo probes.