final changes done before unleashing it on the whole dataset