Revert "Remove time take 2"

This reverts commit 7e990e2a3c104beca03b3042f26619ff806cad08.