Add code highlighting, fix #24