Use `Int` as count return type

Merge pull request #3 from daehn/feature/set-count-directly

Convert to use Int as count type, make count return non-optional