Add function to format data for running models in segan