Add EOF check