clean up + doc update