Update annotation_to_weight.rb