Enhance code, add method to time the execution of another method