Add badges to models