Added todo elements to controller for final bit of work