The "Codelens" feature for step-by-step program execution only works for the pre-programmed examples. If you want the same visualization for your own programs you can use pythontutor.com