Make argument optional/Fixes adding