Came up with a pretty brute force solution for part 2, but I am tired and I am happy I did it :)