Add unit tests for parser