Added header for training purposes