Removed unneeded dimension from driving_function_imp_wfs.m