add check logic