Updated demo layout to be nicer