Add size to validation_dataset