Move key_setup up