Update main.go

implement time limit