add logic for 4 test