SPARK-658: Adding logging of stage duration