Need to check `inverse_transform(X_trans.tolist())` here