Adding root node support

Todo: Update grammar and readme