Move done() call to inside res so the test executes properly
Add more test functionality