Add comments throughout