small edits
added simple tests and tests output