Add write timeout