One more change, can pass in functions as well.