Update main.c

2 user stories added