add more test cases and profiling annotation