Change Weight.count return value from Map to Integer array