DOC: clean code