When imputing interactions, squares, and so on, transform then impute