Remove function to initially show module weights