CDK-721: Add main class to kite-dataset shell script.