Fix params and update code