add restart feature to lesson