correction to arduino code