Added class references to source code