Python step 4 + comments