Fixed error in time_taken