Add infection counter to vis

- Position of counter non-ideal, can't seem to set it
- Probably best to have updateNode() handle label update via bool param
- Update TODO