added missing header to main.h