Created light/heavy training functions, w_sq matrix