Added button testing code in main.c