Fixed imports in main.h