Added a header tag for training