Add demo, cleanup js and css