Enhancement: add badges