Add a feature to extract runtime of the code and print it after code execution.
Add a feature to extract runtime of the code and print it after code execution.