Added plotting of full state