changes the create_dataset function to a more generic form taking the initial condition as arguments
changes