small changes to main file