Python script updates