[Yujun] Refactored tests and modified readme.