Added set keyspace functionality