Added a couple #ifdefs

Added #ifdef for SLAVE and MASTER in main.c