update to use main