I need help editing some existing code