Updated Python script and REadme