Write function for creating training/test split in loaded data

Created by: MattiasBeming

See #2 (closed) #5 (closed)