fixed weird whitespace issue
added a few tests