Added weight initialization and a visualizer for the network.