Update test cases and added locust example