removed import add_weights