Small clean up, adding require