Add define.h to switch more easily between main functions.