back prop for linear layer is now implemented