Code Generation Assistant

Provide a description of the code you need, and the model will return Python code only.