Add input polling