Modify An Array In Python Problem