Added hidden layer capacity

Not tested yet!