Uppdaterad kod main.c