Refactor of data creation
Corrected instance for train 8