modify method to call main