Edit below code as if was an input