Updated problem 3 script and got pretty neat results