Updated Python script to deal with integers