Update main.c

Added Time-Measurement functionality similar to part_1