added functionality in script