Changed name of weights in _forward