Code cleanup

Change nested stream to method reference