update to call correct function