Add input cleanup