Creation of the weight matrix works :)