I would keep lessons short. At his age owners are lucky to even get any attention at times.
He's learning and his attention span will develop and grow. I believe the best would be to keep it positive, fun for you BOTH!, and end on good notes. Good luck!