modified train.m to include all experiments