Modify scripts for extra features