Add input buffer support