put it in the right function