Add capability to add references between classes