Move output printing to task decorator