I refactor previous commit to fit with unmerged pull request: https://github.com/tmdvs/TDBadgedCell/pull/15/files