Add template code for lab 7