Getting started with writing Code Correctness problems (Python)