Add a tiny bit more socket stuff