simple python code