Almost finished lab 2