added wrapper for model prep