Implemented missing bits to pass tests

Hey Bart. It was great to read through your blog posts after reading Terence Parrs books. I had fun completing the tiny language using the clean new antlr4 style - and even adjusted the grammar a couple of times to use different styles. Given how much you have given me I thought I would contribute back with this pull request. I hope you find it useful.