Signed-off-by: Eric Zhao <sczyh16@gmail.com>
- Code refinement - Add test cases Signed-off-by: Eric Zhao <sczyh16@gmail.com>