Update timetrial.py

Specify n by command line