Updated task module