Add icon badging. Closes #10