ok, I will make the changes and update the PR. thanks for the comments