Extending State.py to include node_gmm