Update main.cpp

add multiple module support