Add more tests on references + fix a bug of a class referencing a class

Add more tests on references + fix a bug of a class referencing a class #199