add option to show badge