Add better logging around feature extraction time