Remove printing

Remove return from task 2, refer to return from task2_impl