add correct functions