Fix code to print timing