I hardcoded the for loop ranges for part 2 because I was tired. Maybe I'll fix one day.