Neural Network overfitting fix
Adding time ability