Adjust how the height parameter is used in the demo