Added tests for Klass as well as organized grammar.py to be neater as well as added more documentation.