Add prop to dynamically assign badges

Add prop to dynamically assign badges #298