Add timeout parameter