Add get_count function.