fixed the second demo a little