simple python code written