Add test count

Fix #5