better dataset generation and headless output