add badge and modify gui