Authoring Code Correctness problems (Python)