Add get_args function