change code based on feedback