key_length.py now works properly