changed size of output weight matrix