update main

sperate it to two files.