Add ability to register a task with just a function.
Add some more tests.