Address review comments
- put formatting of tasks completed back to what it was
- update test to pass return_value as an argument to Mock()