Add main.c for testing the functions