Updated python script to fix bug