Modified params to be same as paper