Add `parse_args` function