Fix snake parameter for heuristics notebook

[colinjfw]

A couple small fixes for the heuristic development notebook.

• Fixes the json generation to use the correct snake id.
• Fixes the visualization parameter to use the snake count.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.