[Feature Request] Step Functions integration

A thing I noticed while using AWS Step Functions, was that I didn't have a way to fetch the task ID from inside my Lambda function.
It would be great if that was available in the context. Even more a closer integration with X-Ray would be very much appreciated.

Thank you and keep up the great work! 👌