Add set_mac function