the initial python code