0.0.1: add count function