Add docstring for set_stp_mode