Test unit

Fix #1