cleanup + update doc