Fix Enumerable#take_while arg handling

Addresses the take_while part of #901