New feature.

I added this additional functionality.