Update

Just noticed I never got around to making the suggested changes