initial python code