Simple python code.