add a return in as well