Updated unit testing to support sauce labs