Add functions lab 2