PS5 - Problem 1 Lab Code