Added some styling to training