@NikoAP77 Well, my code is 1000% correct, I know what I did and I love it 
Please see the below outputs of the above suggested code:
Selection base on buttons:
Selection base on user input:
I even mentioned “If code not run, may be I made some stupid error” but I think I should not even wrote this. You just need to customised as per your need but… Next time please run the code and then provide the feedback please.
Good Luck for life!