added keypoint dataset generation code