Small modification of main file