Refactored First Test Case