removed unecessary code in demo