Don't know how to make takeWhile work nicely...