pylint clean-up
added UML