Change return type
Add ability to measure time taken by solution