Minor change to demo section