Add set_diver
Cosmetics