Add main
remove large dataset and query