made a few nit fixes