Tweak to read N messages or read forever.