Update comments in demo