Complete assignment
Add output