Add count.go and sample file for testing