Realized an error with the che code we wrote in class. Attempting to fix it.