better initial test
added one more test
fixed code example