make changes based off PR feedback