Add lazy loading for subDocument's classes