Added test lines.

This fixes #2