Add functions to main module import