Let count method to take arguments