small adjustment to Main