Adding socket testing