add train.py & wcl_dataset