Add dataset, polish demos