Added output layer in network creation