Add more complete python script