altered for laptop
merge training and test