Add get_valid_key function
Add docs