Updated IAM dataset creation