Updated Trainer to work with levels