Step 8 – Building the controller