Add task module
Fix tests