Updated check_weight function to figure out missing weights