Add tests directory
Fix #36