GitHub #89 - Add tests