Update HW 6

new small changes to how norm.grad is calculated.