ENH: Create function to parse args