some updates to existing script