[Data generation] Simplified data init