Junior python code