Adding types to the main module