correct issue in logic: not changing values used to index data, changing ordering of access
make __main__ better demonstrate functionality