An ex-pen is the best with washable piddle pads. Be sure to use an enzyme cleaner on the spots that he piddles or poos on. He is such a tiny baby yet. He will do just great as he gets older. And do remember, Yorkies are known for their stubbornness.
I never trusted mine running around until they were at least 1 yr. old.