Include momentum distribution in momentum resolution plots