Added huber loss function. Fixed some prediction print-outs.