Is this correct?

Lessons are what we took at the pool to learn to swim better.

Answer: to learn to swim better, we took lessons at the pool.

Yes. Be sure to capitalize the first letter.