Rewrote finding max string length

- Using list comprehensions because Python is nice.