Adding cProfile for checking times of each function calls in our test. What to improve?