Fix Observable.take, add tests