Add tests for the parser