Trying to fix key state display (before lab)