Add dir check