fixing up pr feedback