updating tests code to optional addition