add test code and use selenium