Add findOne method to the query builder