Tidy up parser code