Improve docstring, improve code readability.
Add Vertex class.