Add tests for reader