Added include to main.c for button.h