Update code based on comment