fixed typo
added more testing