Created trainLab2.py