Add the return keyword
Enforce test and invariant block bodies to be compound statements

This increases readability and consistency by enforcing it.
Add invariant and test keywords
Add parsing of test and invariant statements