Finishing up Lab 1 assignment