improved demo and integrated demo into docs