Added test & fixed whitespace