syntax highlighting
add doctest
Merge pull request #2 from livibetter/small-changes

Small changes