Minor bug fix - needed to fix import of gridExtra to ensure visible Decision Tree graphics(?). Other small tweaks