Add more different training, and other features