Added hidden layer option