Edited add comment code