Programming Assignment Creating A Custom Loss Function