Add get_weight