Python code now can handle more than one button via a dictionnary