Code comprehension refers to the ability to read, understand, and explain code effectively. While students may successfully complete introductory programming assignments, they often do so without fully grasping the underlying code, which can lead to difficulties in more advanced classes. This work presents our approach to addressing this challenge through targeted interventions that assess students’ comprehension using personalised, AI-generated questions. We share insights from a pilot study conducted to gather initial student feedback on the effectiveness of this method.