Some code cleanup

SetupButton function overloaded for MenuButton type.

Change-Id: Ib171c7590a59ad360dd8ebab1039d3eba5d565ef
All templates option added

Change-Id: I622d5d6fd7fa7be474fa56abf67a1a62fddbcf60