Added tests covering the behavior from #4.