US-5: Step 2. code implementation