Add functions for argument handling