Got weights format correct