Updates tests

Methods were not exposing "count" value in returned hash. This has been
corrected by returning the full response that comes from the API.