small changes to main