added the new functionality