Added Take and TakeWhile tests