small fix
Comments and cleanup.