Add samples loader (train.py)