Python script to