first draft of adding a second hidden layer