Added weights to loss
Typo