refactor: remove counts to make it simply