better dataset python file creation