Adding training & evaluation code.