Adding count option