Update code so we can add more icons in the future