add some modification to main module