Update TestCustomProtocolSimulation.scala

Fixed start end time capture.