Improve CodeLesson#edit