a small nitpick here can you define that just after the imports instead of the middle of the function?
Thanks!