Adding arg parse