C4W1 Assignment train_val_split problem

@Pratham_Pal1
Here’s a hint to fix train_val_split:
Please use function parameters and local variables instead of global variables when possible.