Private 7th place solution | Kaggle