Update to work with iTerm