Added test and train images