set up to include training option