Add argument parsing