Improving Expressivity of Graph Neural Networks using Localization