added rack attribute as an optional parameter
diff --git a/library/cassandra_hosts.py b/library/cassandra_hosts.py
index 5862d44..c15f669 100644
--- a/library/cassandra_hosts.py
+++ b/library/cassandra_hosts.py
@@ -39,7 +39,7 @@
                 if 'rack' in hostvar:
                     rack = hostvar['rack']
                 else:
-                    rack = '1'
+                    rack = 1
             cassandra_ip_map = cassandra_ip_mappings[cassandra_group_name]
             cassandra_ip_map[ds_ip] = { 'private_ip': private_ip, 'rack': rack}
     return cassandra_ip_mappings