Add badge editin feature