fixed shapes output in task 3