Added main module for nicer usage