Adding code for the optional arguments experiment