added early stopping
included necessary params in config