test files. modify weights to be independent of num of modules
remove temp