add test cases
fix UI