function search
    if test (count $argv) -lt 1
        echo "Usage: search <string> [directory]"
        return 1
    end

    set query $argv[1]
    set dir "."

    if test (count $argv) -ge 2
        set dir $argv[2]
    end

    rg "$query" "$dir"
end