Modified demo file and added styles to make it look good