Small update

This is a small update to my function