modified soln but TLE