Update main py