Trying to finish lab 2