Added train() without validation data