Small fixes in demo and removed console logs