minor cleanup

Just cleaned up the code a bit as suggested on GoogleCloudPlatform/endpoints-proto-datastore/pull/82