Step 5 - add controllers