Add argument to handle options