updated script to new source files