Step 4: Add Models and Helpers