16-NAME1.py a little bit simplified but not optimized, editorial solution