Replace states dictionary with pressed_keys set