Bug fixed related to data class

A bug was fixed related to imputation when the formula method is used
with \code{train}