demo is working, tweaked some code so that button toggles