nit and added comments