Step 5: Unit testing