add correct calculation