Implementing dropout to avoid overfitting