Finish mist_keys module refactoring