Updated to match the functionality of the add_derived_column function.