Add support for pit.only

This is to match it.only API.