adding timeout to tests