Updated driving_function_imp_wfs_pw()