Add badge code