Set main file properly