Move comment UI to lesson page