Cleaned up
Found take_while method in api. really usefull :D