made minor changes to main