Refreshing the 2D demo to be nicer and the network to be changeable.