[DOC] Clean up readme