Change the way menu component works