added option for equalization in the kaggle code
added argument for equalization with kaggle as well