Unit Test: Move the test steps to the right place, and Correct the comments.