Added the ability to initialize the network with random weights