hw1: add output function