add return in run function