Added hidden layer and cleaned up screen