For simplicity, we were using Strings as the keys, but we should be allowing any key type. This issue should be solved when #2 is done.