Working on first part of lab 2