Step 2.2.1 complete, add second menu to choose training, validation or test