Started improving the training activity gui