Add main py file