Documentation in main.c updated