Fixed code syntaxe and adde one more test.