Add logic for the class Argument