Add dir argument