Adding take() method to Stream