Add support for adding related for added model